package com.baiyu.common.utils.word;

import com.baiyu.common.utils.file.FileUtils;
import freemarker.template.Configuration;
import freemarker.template.Template;
import org.springframework.util.ClassUtils;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.URLEncoder;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Map;

public class WordUtils {
    //配置信息,代码本身写的还是很可读的,就不过多注解了21
    private static Configuration configuration = null;
    //这里注意的是利用WordUtils1的类加载器动态获得模板文件的位置
//    private static final String templateFolder = WordUtils1.class.getClassLoader().getResource("/").getPath();
    private static String templateFolder=null;

    static {
        try {
            configuration = new Configuration();
            configuration.setDefaultEncoding("utf-8");
            templateFolder = ClassUtils.getDefaultClassLoader().getResource("").getPath()+"templates/";;
            templateFolder = FileUtils.toUNIXpath(templateFolder);
            configuration.setDirectoryForTemplateLoading(new File(templateFolder));
            System.out.println(templateFolder);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    private WordUtils() {
        throw new AssertionError();
    }

    public static void exportMillCertificateWord(HttpServletRequest request, HttpServletResponse response, Map map, String title, String ftlFile) throws IOException {
        long time1 = new Date().getTime();
        Template freemarkerTemplate = configuration.getTemplate(ftlFile);
        File file = null;
        InputStream fin = null;
        ServletOutputStream out = null;
        try {
            // 调用工具类的createDoc方法生成Word文档
            file = createDoc(map, freemarkerTemplate);
            fin = new FileInputStream(file);

            response.setCharacterEncoding("utf-8");
            response.setContentType("application/msword");
            // 设置浏览器以下载的方式处理该文件名
            String fileName = title + ".doc";
            response.setHeader("Content-Disposition", "attachment;filename="
                    .concat(String.valueOf(URLEncoder.encode(fileName, "UTF-8"))));

            out = response.getOutputStream();
            byte[] buffer = new byte[1024];  // 缓冲区
            int bytesToRead = -1;
            // 通过循环将读入的Word文件的内容输出到浏览器中
            while ((bytesToRead = fin.read(buffer)) != -1) {
                out.write(buffer, 0, bytesToRead);
            }
        }catch (Exception e){

        }finally {
            if (fin != null) fin.close();
            if (out != null) out.close();
            if (file != null) file.delete(); // 删除临时文件
        }
//        System.out.println(new Date().getTime()-time1);
    }

    private static File createDoc(Map<?, ?> dataMap, Template template) {
        String name = "file.doc";
        File f = new File(name);
        Template t = template;
        try {
            // 这个地方不能使用FileWriter因为需要指定编码类型否则生成的Word文档会因为有无法识别的编码而无法打开
            Writer w = new OutputStreamWriter(new FileOutputStream(f), "utf-8");
            t.process(dataMap, w);
            w.close();
        } catch (Exception ex) {
            ex.printStackTrace();
            throw new RuntimeException(ex);
        }
        return f;
    }


    /**
     * 工具：根据身份证的号码算出当前身份证持有者的性别和年龄--18位身份证
     */
    public int age(String CardCode) {
        if (CardCode.length() == 18) {
            String year = CardCode.substring(6).substring(0, 4);// 得到年份
            String yue = CardCode.substring(10).substring(0, 2);// 得到月份
            String day = CardCode.substring(12).substring(0, 2);//得到日
            String sex;
            if (Integer.parseInt(CardCode.substring(16).substring(0, 1)) % 2 == 0) {// 判断性别
                sex = "女";
            } else {
                sex = "男";
            }
            Date date = new Date();// 得到当前的系统时间
            SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd");
            String fyear = format.format(date).substring(0, 4);// 当前年份
            String fyue = format.format(date).substring(5, 7);// 月份
            String fday = format.format(date).substring(8, 10);//日期
            int age = 0;
            if (Integer.parseInt(yue) <= Integer.parseInt(fyue)) { // 当前月份大于用户出身的月份表示已过生
                if (Integer.parseInt(yue) == Integer.parseInt(fyue)) {
                    if (Integer.parseInt(day) <= Integer.parseInt(fday)) {// 当前日期大于用户出身的月份表示已过生
                        age = Integer.parseInt(fyear) - Integer.parseInt(year);
                    } else {
                        age = Integer.parseInt(fyear) - Integer.parseInt(year) - 1;
                    }
                } else {
                    age = Integer.parseInt(fyear) - Integer.parseInt(year);
                }
            } else {// 当前用户还没过生
                age = Integer.parseInt(fyear) - Integer.parseInt(year) - 1;
            }
            return age;
        } else {
            return 0;
        }
    }

}
